package leetcode.code0304;

import leetcode.helper.HelpDebug;

public class NumMatrix {
	int[][] sums;

	public NumMatrix(int[][] matrix) {
		int m = matrix.length;
		int n = matrix[0].length;
		sums = new int[m + 1][n + 1];
		for (int i = 1; i <= m; i++) {
			for (int j = 1; j <= n; j++) {
				sums[i][j] = sums[i - 1][j] + matrix[i - 1][j - 1];
			}
		}
		for (int i = 1; i <= m; i++) {
			for (int j = 1; j <= n; j++) {
				sums[i][j] += sums[i][j - 1];
			}
		}
	}

	// row1 col1 row1 col2
	// row2 col1 row2 col2
	public int sumRegion(int row1, int col1, int row2, int col2) {
		HelpDebug.print(sums);
		return sums[row2 + 1][col2 + 1] - sums[row2 + 1][col1] - sums[row1][col2 + 1] + sums[row1][col1];
	}

	public static void main(String[] args) {
		NumMatrix numMatrix = new NumMatrix(
				HelpDebug.str2array1("[[3,0,1,4,2],[5,6,3,2,1],[1,2,0,1,5],[4,1,0,1,7],[1,0,3,0,5]]"));
		numMatrix.sumRegion(2, 1, 4, 3); // return 8 (红色矩形框的元素总和)
		numMatrix.sumRegion(1, 1, 2, 2); // return 11 (绿色矩形框的元素总和)
		numMatrix.sumRegion(1, 2, 2, 4); // return 12 (蓝色矩形框的元素总和)

	}

}
